"""
!/usr/bin/env python
-*- coding: utf-8 -*-
@Author  : MarkJia
@Site    : 
@File    : 7、第一个例子：百度python.py
"""
import time

from selenium import webdriver
from selenium.webdriver.common.by import By


class Testpython:

    # 用例前置：打开浏览器
    def setup_class(self):
        self.driver = webdriver.Chrome()
        self.driver.get('https://www.baidu.com')
        self.driver.maximize_window()
        # 设置隐式等待 : 10s
        self.driver.implicitly_wait(10)

    # 用例后置：关闭浏览器
    def teardown_class(self):
        self.driver.quit()

    def test_python(self):
        # 用 By.ID 方法定位到百度搜索框，清空内容
        self.driver.find_element(By.ID, 'kw').clear()
        # 输入 python
        self.driver.find_element(By.XPATH, "//input[@id='kw']").send_keys('python')
        # 定位 百度一下 按钮，并点击
        self.driver.find_element(By.CSS_SELECTOR, '#su').click()
        time.sleep(2)

        # 断言 搜索标题是不是  '百度一下，你就知道'
        assert self.driver.title == 'python_百度搜索'
